#include <stdio.h>
#include "main.h"
#include "menu.h"
#include "bsp_debug_usart.h"
#include "bsp_led.h"  
#include "common.h"

void SystemClock_Config(void);

extern UART_HandleTypeDef UartHandle;
extern pFunction JumpToApplication;
extern uint32_t JumpAddress;

//static void IAP_Init(void);
void SystemClock_Config(void);

static void Drv_Init(void)
{
	HAL_Init();

  SystemClock_Config();
	
	Led_Gpio_Init();
	 
	FLASH_If_Init();

	//IAP_Init();
	Debug_Usart_Init();
}

/**
  * @brief  The application entry point.
  * @retval int
  */
int main(void)
{
  uint8_t key = 0;
	
	Drv_Init();
	
	HAL_Delay(5000);
	
	Serial_PutString("\r\n=================== Main Menu ============================\r\n\n");
	Serial_PutString("  Download image to the internal Flash ----------------- 1\r\n\n");
	Serial_PutString("  Upload image from the internal Flash ----------------- 2\r\n\n");
	Serial_PutString("  Execute the loaded application ----------------------- 3\r\n\n");
 	//printf("\rhit any key to stop autoboot:  ");
	
	HAL_UART_Receive(&UartHandle, &key, 1, 10000);
	if(key == '1')
	{
	  Main_Menu ();
	}
	else
	{
		Serial_PutString("Start program execution......\r\n\n");
		/* execute the new program */
		JumpAddress = *(__IO uint32_t*) (APPLICATION_ADDRESS + 4);
		/* Jump to user application */
		JumpToApplication = (pFunction) JumpAddress;
		/* Initialize user application's Stack Pointer */
		__set_MSP(*(__IO uint32_t*) APPLICATION_ADDRESS);
		JumpToApplication();
	}
	
  while (1)
  {

  }

}

/**
  * @brief System Clock Configuration
  * @retval None
  */
void SystemClock_Config(void)
{
  RCC_OscInitTypeDef RCC_OscInitStruct = {0};
  RCC_ClkInitTypeDef RCC_ClkInitStruct = {0};

  /** Initializes the RCC Oscillators according to the specified parameters
  * in the RCC_OscInitTypeDef structure.
  */
  RCC_OscInitStruct.OscillatorType = RCC_OSCILLATORTYPE_HSE;
  RCC_OscInitStruct.HSEState = RCC_HSE_ON;
  RCC_OscInitStruct.HSEPredivValue = RCC_HSE_PREDIV_DIV1;
  RCC_OscInitStruct.HSIState = RCC_HSI_ON;
  RCC_OscInitStruct.PLL.PLLState = RCC_PLL_ON;
  RCC_OscInitStruct.PLL.PLLSource = RCC_PLLSOURCE_HSE;
  RCC_OscInitStruct.PLL.PLLMUL = RCC_PLL_MUL9;
  if (HAL_RCC_OscConfig(&RCC_OscInitStruct) != HAL_OK)
  {
    Error_Handler();
  }

  /** Initializes the CPU, AHB and APB buses clocks
  */
  RCC_ClkInitStruct.ClockType = RCC_CLOCKTYPE_HCLK|RCC_CLOCKTYPE_SYSCLK
                              |RCC_CLOCKTYPE_PCLK1|RCC_CLOCKTYPE_PCLK2;
  RCC_ClkInitStruct.SYSCLKSource = RCC_SYSCLKSOURCE_PLLCLK;
  RCC_ClkInitStruct.AHBCLKDivider = RCC_SYSCLK_DIV1;
  RCC_ClkInitStruct.APB1CLKDivider = RCC_HCLK_DIV2;
  RCC_ClkInitStruct.APB2CLKDivider = RCC_HCLK_DIV1;

  if (HAL_RCC_ClockConfig(&RCC_ClkInitStruct, FLASH_LATENCY_2) != HAL_OK)
  {
    Error_Handler();
  }
}

/**
  * @brief  This function is executed in case of error occurrence.
  * @retval None
  */
void Error_Handler(void)
{
  /* USER CODE BEGIN Error_Handler_Debug */
  /* User can add his own implementation to report the HAL error return state */
  __disable_irq();
  while (1)
  {
  }
  /* USER CODE END Error_Handler_Debug */
}


